import torch
print(torch.__version__)
print(torch.version.cuda)           # PyTorch 编译的 CUDA 版本
print(torch.cuda.is_available())    # True
print(torch.cuda.get_device_name(0))